# -*- coding: utf-8 -*-
# @Time    : 2022/10/22 13:13
# @Author  : 限量老
'''
自定义的分页组件,以后如果想要使用谤个分页组件,需要做如下几件事:
在视图函数中:
def pretty_list(request):
    1.根据自己的情况去筛选自己的数据
    queryset = models.PrettyNum.objects.filter(**data_dict).order_by('-level')
    2.实例化分页对象
    page_object=Pagination(request,queryset)
    context={'search_data': search_data,
             'queryset': page_object.page_queryset,#分完页的数据
             'page_string': page_object.html()}#生成的页码
    return render(request, 'pretty_list.html',context)

在html中:
<tbody>
	{% for obj in queryset %}
	    {{obj.xx}}
	    {{obj.YY}}
    {% endfor %}
</tbody>

    <ul class="pagination" style="float: left">
        {{ page_string }}
    </ul>
'''
from django.utils.safestring import mark_safe


class Pagination(object):
    def __init__(self,request,queryset,page_size=10,page_parm='page',plus=5):
        """
        :param request:请求的对象
        :param queryset:符合条件的数据(根据这个数据给他进行分页处理)
        :param page_size:每页显示多少条数据
        :param page_parm:在URL中传递的获取分页的参数,例如:/pretty/list/?page=12
        :param plus:显示当前页的前或后几页(页码)
        """
        page = request.GET.get(page_parm, "1")

        import copy
        from django.http.request import QueryDict
        query_dict = copy.deepcopy(request.GET)  # 深拷贝一份从页面URL框接收的数据:?q=888&page=20
        query_dict._mutable = True  # 修改原码中设定为True
        self.query_dict=query_dict
        self.page_parm=page_parm
        if page.isdecimal():#判断输入的是否数字
            page=int(page)
        else:
            page='1'#如果不是就显示第一页
        self.page=page
        self.page_size=page_size

        self.start = (page - 1) * page_size
        self.end = page * page_size

        self.page_queryset=queryset[self.start: self.end]


        total_count = queryset.count()  # 符合条件的总条数
        # 计算出总页码
        total_page_count, div = divmod(total_count, page_size)  # 如果有余数加一页
        if div:
            total_page_count += 1
        self.total_page_count=total_page_count

        self.plus=plus

    def html(self):
        # 计算出显示当前面的前5页,后5页(page就是当前页)

        if self.total_page_count <= 2 * self.plus + 1:
            # 数据库数据比较少
            start_page, end_page = 1, self.total_page_count
        else:
            # 数据库数据比较多
            if self.page <= self.plus:  # 当前而小于5
                start_page, end_page = 1, 2 * self.plus + 1
            else:
                # 当前页>5
                if (self.page + self.plus) > self.total_page_count:
                    start_page, end_page = self.total_page_count - 2 * self.plus, self.total_page_count
                else:
                    start_page = self.page - self.plus
                    end_page = self.page + self.plus

        page_str_list = []

        # 首页:
        self.query_dict.setlist(self.page_parm, [1])  # 带着原来的搜索条件,加上自己的搜索条件
        page_str_list.append(f'<li><a href="?{self.query_dict.urlencode()}">首页</a></li>')
        # 上一页:
        if self.page > 1:
            self.query_dict.setlist(self.page_parm, [self.page - 1])  # 带着原来的搜索条件,加上自己的搜索条件
            prev = f'<li><a href="?{self.query_dict.urlencode()}">上一页</a></li>'

        else:
            self.query_dict.setlist(self.page_parm, [1])  # 带着原来的搜索条件,加上自己的搜索条件
            prev = f'<li><a href="?{self.query_dict.urlencode()}">上一页</a></li>'  # 如果当前页等于1,就显示当前页
        page_str_list.append(prev)

        # 页面:
        for i in range(start_page, end_page + 1):
            self.query_dict.setlist(self.page_parm, [i])
            if i == self.page:  # 如果是当前页,定位当前页
                ele = f'<li class="active"><a href="?{self.query_dict.urlencode()}">{i}</a></li>'  # 把html语句在这里循环

            else:
                ele = f'<li><a href="?{self.query_dict.urlencode()}">{i}</a></li>'  # 把html语
            page_str_list.append(ele)

        # 下一页:
        if self.page < self.total_page_count:
            self.query_dict.setlist(self.page_parm, [self.page + 1])
            prev = f'<li><a href="?{self.query_dict.urlencode()}">下一页</a></li>'

        else:
            self.query_dict.setlist(self.page_parm, [self.total_page_count])
            prev = f'<li><a href="?{self.query_dict.urlencode()}">下一页</a></li>'  # 如果当前页等于1,就显示当前页
        page_str_list.append(prev)

        # 尾页:
        self.query_dict.setlist(self.page_parm, [self.total_page_count])
        page_str_list.append(f'<li><a href="?{self.query_dict.urlencode()}">尾页</a></li>')

        # 跳转页面:
        search_string = '''
        	<li>
            <form method="get" style="float:left;margin-left: -1px;">
            <input type="text" style="position:relative; float: left;width:80px; border-radius:0; display: inline-block;"
                name="page" class="form-control" placeholder="页码">
            <button style="border-radius: 0;" class="btn btn-default" type="submit">跳转</button>
            </form>
            </li>
            '''
        page_str_list.append(search_string)

        # 从view向template传递HTML字符串的时候，django默认不渲染此HTML，原因是为了防止这段字符串里面有恶意攻击的代码
        page_string = mark_safe(''.join(page_str_list))  # 安全机制
        return page_string